@echo off
chcp 65001 >nul
echo ====================================
echo 设置 Python 3.11.9 for pandasai
echo ====================================
echo.

REM 设置本地 Python 版本为 3.11.9
echo [1/4] 设置项目本地 Python 版本...
%USERPROFILE%\.pyenv\pyenv-win\bin\pyenv.bat local 3.11.9
if %errorlevel% neq 0 (
    echo 错误：设置本地版本失败
    pause
    exit /b 1
)
echo ✓ 已设置为 Python 3.11.9
echo.

REM 验证 Python 版本
echo [2/4] 验证 Python 版本...
%USERPROFILE%\.pyenv\pyenv-win\versions\3.11.9\python.exe --version
echo.

REM 创建虚拟环境
echo [3/4] 创建虚拟环境 venv_pandasai...
if exist venv_pandasai (
    echo 虚拟环境已存在，跳过创建
) else (
    %USERPROFILE%\.pyenv\pyenv-win\versions\3.11.9\python.exe -m venv venv_pandasai
    if %errorlevel% neq 0 (
        echo 错误：创建虚拟环境失败
        pause
        exit /b 1
    )
    echo ✓ 虚拟环境创建成功
)
echo.

REM 激活虚拟环境并安装 pandasai
echo [4/4] 安装 pandasai...
call venv_pandasai\Scripts\activate.bat
python -m pip install --upgrade pip
pip install pandasai
if %errorlevel% neq 0 (
    echo 错误：安装 pandasai 失败
    pause
    exit /b 1
)
echo ✓ pandasai 安装成功
echo.

echo ====================================
echo 安装完成！
echo ====================================
echo.
echo 使用说明：
echo 1. 激活虚拟环境：
echo    venv_pandasai\Scripts\activate
echo.
echo 2. 运行你的脚本：
echo    python your_script.py
echo.
echo 3. 退出虚拟环境：
echo    deactivate
echo.
pause

